package com.yx.changdao.common.utils;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.lang.reflect.Field;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.net.URLEncoder;
import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;

/**
 * @Description: <br>
 * CSV 文件导出工具类
 *
 * <p><br>
 * @Auther: Zhangjiashun
 * @create 2019/11/19 9:50
 */
public class ExportCSVUtils {

    public static final Logger log = LoggerFactory.getLogger(ExportCSVUtils.class);

    // CSV文件列分隔符        \t 是为了将所有数字类型字符转化为字符串。
    private static final String CSV_COLUMN_SEPARATOR = ",\t";
    // CSV文件换行符
    private static final String CSV_RN = "\r\n";

    /**
     * CSV文件导出
     *
     * @param fileName 输出文件名
     * @param response 响应 response
     * @param dataList 集合数据列表 包含内容为多个map（数据键：数据）
     * @param colNames 表头
     * @param mapKey   集合数据中的键 影响数据列输出顺序。
     * @return 是否导出成功
     */
    public static Boolean export(
            String fileName,
            HttpServletResponse response,
            List<String> colNames,
            List<String> mapKey,
            List<Map<String, Object>> dataList) {

        if (null == dataList || null == colNames || null == mapKey) {
            return false;
        }

        StringBuilder buf = new StringBuilder();

        // 输出列头
        for (String col : colNames) {
            buf.append(col).append(CSV_COLUMN_SEPARATOR);
        }

        // 下一行
        buf.append(CSV_RN);

        // 输出数据
        for (Map<String, Object> data : dataList) {
            for (String key : mapKey) {
                buf.append(data.get(key)).append(CSV_COLUMN_SEPARATOR);
            }
            buf.append(CSV_RN);
        }

        return csvResponseSetProperties(fileName, buf.toString(), response);
    }

    /**
     * CSV文件导出 输出的数据列顺序为 实体类中属性的顺序
     *
     * @param fileName   文件名
     * @param response   响应response
     * @param clazz      导出数据类型
     * @param colNames   表头
     * @param dataList   集合数据列表 包含多个clazz类型的值
     * @param excludeKey 排除的属性
     * @return 是否导出成功
     */
    public static Boolean export(
            String fileName,
            HttpServletResponse response,
            Class<?> clazz,
            List<String> colNames,
            List<?> dataList,
            List<String> excludeKey) {

        return export(fileName, response, clazz, colNames, dataList, excludeKey, "yyyy-MM-dd");
    }

    /**
     * CSV文件导出 输出的数据列顺序为 实体类中属性的顺序
     *
     * @param fileName           文件名
     * @param response           响应response
     * @param clazz              导出数据类型
     * @param colNames           表头
     * @param dataList           集合数据列表 包含多个clazz类型的值
     * @param excludeKey         排除的属性
     * @param dateFormatPartener 日期格式化方式
     * @return 是否导出成功
     */
    public static Boolean export(
            String fileName,
            HttpServletResponse response,
            Class<?> clazz,
            List<String> colNames,
            List<?> dataList,
            List<String> excludeKey,
            String dateFormatPartener) {

        if (null == dataList || null == colNames) {
            return false;
        }

        StringBuffer content = new StringBuffer();

        // 输出列头
        colNames.forEach(item -> content.append(item).append(CSV_COLUMN_SEPARATOR));
        // 下一行
        content.append(CSV_RN);

        // 获取类所有成员属性
        Field[] declaredFields = clazz.getDeclaredFields();
        List<String> mapKeys = new ArrayList<>();
        for (Field field : declaredFields) {
            field.setAccessible(true);
            String name = field.getName();
            // 排除静态属性 和 传入需要排除的属性
            if (Modifier.isStatic(field.getModifiers())
                    || excludeKey == null
                    || excludeKey.contains(name)) {
                continue;
            }
            mapKeys.add(name);
        }

        // 组装数据
        for (Object obj : dataList) {
            for (String col : mapKeys) {
                // 获取拼接get方法
                String methodName = "get" + col.substring(0, 1).toUpperCase() + col.substring(1);
                Object value;
                try {
                    Method method = clazz.getDeclaredMethod(methodName);
                    //            method.setAccessible(true);
                    value = method.invoke(obj);
                } catch (InvocationTargetException | NoSuchMethodException | IllegalAccessException e) {
                    // private修饰的get方法跳过、 方法内部异常 跳过、 没有 get 方法跳过
                    break;
                }

                // 将日期类型格式化
                if (!StringUtils.isBlank(dateFormatPartener)) {
                    if (value instanceof Date) {
                        SimpleDateFormat sdf = new SimpleDateFormat(dateFormatPartener);
                        value = sdf.format(value);
                    }
                }

                content.append(value);
                content.append(CSV_COLUMN_SEPARATOR);
            }
            content.append(CSV_RN);
        }

        return csvResponseSetProperties(fileName, content.toString(), response);
    }

    /**
     * 设置csv响应头
     *
     * @param fileName 文件名
     * @param response 响应resonse
     */
    private static Boolean csvResponseSetProperties(
            String fileName, String content, HttpServletResponse response) {
        // 设置文件后缀
        String fn = fileName + ".csv";
        // 读取字符编码
        String charset = "UTF-8";

        // 设置响应
        response.setContentType("application/shlnd.ms-excel;charset=utf-8");
        response.setCharacterEncoding(charset);
        response.setHeader("Pragma", "public");
        response.setHeader("Cache-Control", "max-age=30");
        try {
            response.setHeader(
                    "Content-Disposition", "attachment; filename=" + URLEncoder.encode(fn, charset));
        } catch (UnsupportedEncodingException e) {
            log.error("转码异常 = ", e);
        }

        ServletOutputStream os;
        try {
            os = response.getOutputStream();
            // 写出响应
            os.write(content.getBytes("UTF-8"));
            os.flush();
            return true;
        } catch (IOException e) {
            e.printStackTrace();
        }
        return false;
    }
}
